import time
from selenium.webdriver.common.by import By

# 页面操作的工具类
class PageTools:
    def __init__(self,driver):  # 初始化类
        self.driver = driver           # 接收 drive 对象
        driver.maximize_window()       # 窗口最大化
        driver.implicitly_wait(10)     # 隐式等待

    def visit(self,url):    # 访问链接地址
        self.driver.get(url)
        time.sleep(1)

    def by_css(self,css):   # 以css方式获取元素定位
        return self.driver.find_element(By.CSS_SELECTOR,css)

    def by_link(self,link):   # 以link链接方式获取元素定位
        return self.driver.find_element(By.PARTIAL_LINK_TEXT,link)

    def quit(self):    # 关闭浏览器驱动
        self.driver.quit()

    def refresh(self):
        self.driver.refresh()

    def del_cookie(self):
        self.driver.delete_all_cookies()